"""
初始化Deepseek模型
安装必要的库
pip install -U langchain-deepseek
pip install python-dotenv
创建.env文件，将Deepseek API Key添加到文件中
DEEPSEEK_API_KEY=your_api_key_here
"""


from langchain_deepseek import ChatDeepSeek
from langchain_core.prompts import ChatPromptTemplate
from dotenv import load_dotenv
import os
# 加载环境变量
load_dotenv()

# 获取环境变量
api_key = os.getenv("DEEPSEEK_API_KEY")

# 初始化模型
llm = ChatDeepSeek(
    model="deepseek-chat",
    api_key=api_key,
    temperature=0.5,
    max_tokens=1024,
    max_retries=2
)

res = llm.invoke("历史上真的有郭靖这个人物吗？")

print(res)

print("\n","------------------")

print(res.content)
